package util;

import org.sqlite.SQLiteConfig;
import org.sqlite.SQLiteDataSource;

import javax.sql.DataSource;
import java.io.File;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import java.net.URLDecoder;
import java.sql.*;

/**
 * @author Dianzhi Han
 * @version 1.0
 * @description:
 * SQLite数据库的工具类，创建数据源，创建数据库的连接
 * 只向外部提供SQLite数据库的连接即可，数据源不提供(封装在工具类的内部)
 * 无论是哪种关系型数据库，操作的流程都是JDBC四步走
 * @date 2022/7/25 11:13
 */
public class DBUtil {
    // 单例数据源
    private volatile static DataSource DATASOURCE;
    // 单例数据库连接
    private volatile static Connection CONNECTION;
    // 获取数据源方法，使用double-check单例模式获取数据源对象
    private static DataSource getDataSource() {
        if (DATASOURCE == null) {
            // 多线程场景下，只有一个线程能进入同步代码块
            synchronized (DBUtil.class) {
                if (DATASOURCE == null) {
                    // SQLite没有账户密码，只需要配置日期格式即可
                    SQLiteConfig config = new SQLiteConfig();
                    config.setDateStringFormat(Util.DATE_PATTERN);
                    DATASOURCE = new SQLiteDataSource(config);
                    // 配置数据源的URL是SQLite子类独有的方法，因此向下转型
                    ((SQLiteDataSource) DATASOURCE).setUrl(getUrl());
                }
            }
        }
        return DATASOURCE;
    }

    /**
     * 配置SQLite数据库的地址
     * mysql: jdbc:mysql://127.0.0.1:3306/数据库名称？
     * 对于SQLite数据库来说，没有服务端和客户端，因此只需要指定SQLite数据库的
     * 地址即可
     * @return
     */
    private static String getUrl() {
        // 获取编译文件夹路径
        // 默认为target/classes/
//        URL classesPathURL = DBInit.class.getClassLoader().getResource("./");
//        System.out.println("classesPathURL为 : " + classesPathURL);
        // 如果直接执行main方法则会得到当前class的绝对路径。
        // 如果封装在jar包里面执行jar包那么会得到当前jar包的绝对路径
        String classesPath = DBUtil.class.getProtectionDomain()
                .getCodeSource().getLocation().getFile();
        File classesDir = null;
        try {
            classesDir  = new File(URLDecoder.decode(classesPath, "UTF-8"));
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException(e);
        }
//        System.out.println("classesPath : " + classesPath);
        // 获取classes的父路径，即target根目录
        String dir = classesDir.getParent();
//        System.out.println("dir : " + dir);
        // 在target根目录下创建数据库
        String url = "jdbc:sqlite://" + dir + File.separator + "search_everything.db";
        System.out.println("获取数据库连接Url为:" + url);
        return url;
    }

    /**
     * 多线程场景下，SQLite要求多个线程使用同一个连接进行处理
     * @return
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        if (CONNECTION == null) {
            synchronized (DBUtil.class) {
                if (CONNECTION == null) {
                    CONNECTION = getDataSource().getConnection();
                }
            }
        }
        return CONNECTION;
    }

    public static void main(String[] args) throws SQLException {
        System.out.println(getConnection());
    }

    public static void close(Statement statement) {
        if (statement != null) {
            try {
                statement.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
    }

    public static void close(PreparedStatement ps, ResultSet rs) {
        close(ps);
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
    }
}